package com.atguigu.algorithm.kmp;

import java.util.Arrays;

/**
 * KMP算法
 *
 * @author xr-8
 * @date 2023-01-04
 * @time 23:57
 */
public class KMPAlgorithm {

    public static void main(String[] args) {
        String str1 = "BBC ABCDAB ABCDABCDABDE";
        String str2 = "ABCDABD";
        //String str2 = "BBC";

        int[] next = kmpNext("ABAB"); //[0, 1, 2, 0]
        System.out.println("next=" + Arrays.toString(next));

//        int index = kmpSearch(str1, str2, next);
//        System.out.println("index=" + index); // 15了
    }

    /**
     * kmp搜索算法
     * @param src 源字符串
     * @param dest 子串
     * @param next 部分匹配表, 是子串对应的部分匹配表
     * @return 如果是-1就是没有匹配到，否则返回第一个匹配的位置
     */
    public static int kmpSearch(String src, String dest, int[] next) {
        //遍历
        for(int i = 0, j = 0; i < src.length(); i++) {
            //需要处理 str1.charAt(i) ！= str2.charAt(j), 去调整j的大小
            //KMP算法核心点
            while( j > 0 && src.charAt(i) != dest.charAt(j)) {
                j = next[j-1];
            }
            if(src.charAt(i) == dest.charAt(j)) {
                j++;
            }
            if(j == dest.length()) {//找到了 // j = 3 i
                return i - j + 1;
            }
        }
        return  -1;
    }

    /**
     * 获取一个字符串（子串）的部分匹配值表
     * AABCAAAD、AABAAAB、ADAAD、ABAB
     * @param dest
     * @return
     */
    public static int[] kmpNext(String dest) {
        int[] next = new int[dest.length()];  //创建一个next 数组保存部分匹配值
        next[0] = 0; //如果字符串是长度为1 部分匹配值就是0
        for(int j = 1, k = 0; j < dest.length(); j++) {
            //当dest.charAt(i) != dest.charAt(j) ，我们需要从next[j-1]获取新的j
            //直到我们发现 dest.charAt(i) == dest.charAt(j) 或者j<=0 时 才退出
            while(k > 0 && dest.charAt(j) != dest.charAt(k)) {//这时kmp算法的核心点
                k = next[k-1];
            }
            //当dest.charAt(i) == dest.charAt(j) 满足时，部分匹配值就是+1
            if(dest.charAt(j) == dest.charAt(k)) {
                k++;
            }
            next[j] = k;
        }
        return next;
    }
}
